# -*- coding:utf-8 -*-
"""
作者:13505
日期:2021年03月13日
"""

import requests
import bs4


def open_url(url):
    # 设置代理
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36 SE 2.X MetaSr 1.0"
    }

    res = requests.get(url, headers=headers)
    return res


def find_movies(res):
    soup = bs4.BeautifulSoup(res.text, "html.parser")

    # 电影名
    movies = []
    targets = soup.find_all("div", class_="hd")
    for each in targets:
        movies.append(each.a.span.text)

    # 评分
    ranks = []
    targets = soup.find_all("span", class_="rating_num")
    for each in targets:
        ranks.append(each.text)

    # 资料
    messages = []
    targets = soup.find_all("div", class_="bd")
    for each in targets[1:]:
        try:
            messages.append(each.p.text.split('\n')[1].strip() + each.p.text.split('\n')[2].strip())
        except Exception as e:
            print(e)
            continue

    result = []
    length = len(movies)
    for i in range(length):
        msg = """
            电影名称：%s
            评分：%s
            资料：%s
        """ % (movies[i], ranks[i], messages[i])
        result.append(msg)

    return result


# 查找一共多少页面
def find_depth(res):
    soup = bs4.BeautifulSoup(res.text, "html.parser")
    depth = soup.find('span', class_="next").previous_sibling.previous_sibling.text
    return int(depth)


def main():
    host = "https://movie.douban.com/top250"
    res = open_url(host)
    depth = find_depth(res)

    result = []
    for i in range(depth):
        url = host + '/?start=' + str(25 * i)
        res = open_url(url)
        result.extend(find_movies(res))

    with open('source/' + '豆瓣TOP250电影.txt', 'w', encoding='utf-8') as f:
        for i in result:
            f.write(i)


if __name__ == '__main__':
    main()
